home *** CD-ROM | disk | FTP | other *** search
/ Libris Britannia 4 / science library(b).zip / science library(b) / DDJMAG / DDJ9310.ZIP / UNDOC.ZIP / TOPICDMP.C < prev    next >
C/C++ Source or Header  |  1993-08-04  |  14KB  |  374 lines
  1. /* TOPICDMP.C -- Dumps topic file from a Windows .HLP or .MVB file.
  2. Pete Davis, August 1993
  3. With some modifications by Andrew Schulman, September 1993
  4. From Dr. Dobb's Journal, October 1993 */
  5.  
  6. #include <time.h>
  7. #include <stdio.h>
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #include <conio.h>
  11. #include <ctype.h>
  12. #include <limits.h>
  13.  
  14. #pragma pack(1)   /* Make sure we get byte alignment */
  15. #include "whstruct.h"
  16. #include "topicdmp.h"
  17.  
  18. HELPHEADER        HelpHeader;        /* Header for Help file.       */
  19. WHIFSBTREEHEADER  WHIFSHeader;       /* WHIFS Header record         */
  20. int               WHIFSLeafOne = -1; /* First WHIFS Leaf Node       */
  21. long              FirstPageLoc;      /* Used by macros for b-trees  */
  22. char              *PhrasesPtr; 
  23. int               Compressed;        /* Is there compression?       */
  24.  
  25. #define MSG(s)              { puts(s); return; }
  26. #define FAIL(s)             { puts(s); exit(1); }
  27.  
  28. #define GET_STRING(f, s) \
  29.     { char *p = (char *)(s); while (*p++ = fgetc(f)) ; *p = 0; }
  30.  
  31. #define BIT_SET(map, bit)   (((map) & (1 << (bit))) ? 1 : 0)
  32.  
  33. // Finds the first leaf in the WHIFS B-Tree
  34. void WHIFSGetFirstLeaf(FILE *HelpFile) {
  35.     int               CurrLevel = 1; /* Current Level in B-Tree */
  36.     BTREEINDEXHEADER  CurrNode;      /* Current Node in B-Tree  */
  37.     int               NextPage = 0;  /* Next Page to go to      */
  38.  
  39.     /* Go to the beginning of WHIFS B-Tree */
  40.     fseek(HelpFile, HelpHeader.WHIFS, SEEK_SET);
  41.     fread(&WHIFSHeader, sizeof(WHIFSHeader), 1, HelpFile);
  42.     FirstPageLoc = HelpHeader.WHIFS + sizeof(WHIFSHeader);
  43.     GotoWHIFSPage(WHIFSHeader.RootPage);  // macro in WHSTRUCT.H
  44.  
  45.     /* Find First Leaf */
  46.     while (CurrLevel < WHIFSHeader.NLevels) {
  47.        fread(&CurrNode, sizeof(CurrNode), 1, HelpFile);
  48.  
  49.        /* Next Page is conveniently the first byte of the page */
  50.        fread(&NextPage, sizeof(int), 1, HelpFile);
  51.        GotoWHIFSPage(NextPage);
  52.        CurrLevel++;
  53.     }
  54.     /* First Leaf page is here */
  55.     WHIFSLeafOne = NextPage;
  56. }
  57.  
  58. // Get a WHIFS file by file number; returns offset and filename
  59. void GetFile(FILE *HelpFile, DWORD Number, long *Offset, char *Name) {
  60.     BTREENODEHEADER CurrentNode;      
  61.     DWORD           CurrPage, counter = 0;
  62.     char            c, TempFile[19];
  63.     
  64.     /* Skip pages we don't need */
  65.     CurrentNode.NextPage = WHIFSLeafOne;
  66.     do {
  67.         CurrPage = CurrentNode.NextPage;
  68.         GotoWHIFSPage(CurrPage);
  69.         fread(&CurrentNode, sizeof(CurrentNode), 1, HelpFile);
  70.         counter += CurrentNode.NEntries;
  71.     } while (counter < Number);
  72.  
  73.     for (counter -= CurrentNode.NEntries; counter <= Number; counter++) {
  74.         GET_STRING(HelpFile, TempFile);
  75.         fread(Offset, sizeof(long), 1, HelpFile);
  76.     }
  77.     strcpy(Name, TempFile);
  78. }
  79.  
  80. // Get SysHeader to see if compression used on help file
  81. void SysLoad(FILE *HelpFile, long FileStart) {
  82.    SYSTEMHEADER    SysHeader;
  83.    FILEHEADER      FileHdr;
  84.    fseek(HelpFile, FileStart, SEEK_SET);
  85.    fread(&FileHdr, sizeof(FileHdr), 1, HelpFile);
  86.    fread(&SysHeader, sizeof(SysHeader), 1, HelpFile);
  87.    if (SysHeader.Revision != 21)
  88.        FAIL("Sorry, TOPICDMP only works with Windows 3.1 help files");
  89.    Compressed = (SysHeader.Flags & COMPRESSION_310) ||
  90.                 (SysHeader.Flags & COMPRESSION_UNKN);
  91. }
  92.  
  93. // Decides how many bytes to read, depending on number of bits set
  94. int BytesToRead(BYTE BitMap) {
  95.     int TempSum, counter;
  96.     TempSum = 8;
  97.     for (counter = 0; counter < 8; counter ++)
  98.        TempSum += BIT_SET(BitMap, counter);
  99.     return TempSum;
  100. }
  101.  
  102. // Decompresses the data using Microsoft's LZ77 derivative.
  103. long Decompress(FILE *HelpFile, long CompSize, char *Buffer) {
  104.    long InBytes = 0;        /* How many bytes read in                    */
  105.    WORD OutBytes = 0;       /* How many bytes written out                */
  106.    BYTE BitMap, Set[16];    /* Bitmap and bytes associated with it       */
  107.    long NumToRead;          /* Number of bytes to read for next group    */
  108.    int  counter, Index;     /* Going through next 8-16 codes or chars    */
  109.    int  Length, Distance;   /* Code length and distance back in 'window' */
  110.    char *CurrPos;           /* Where we are at any given moment          */
  111.    char *CodePtr;           /* Pointer to back-up in LZ77 'window'       */
  112.  
  113.    CurrPos = Buffer;
  114.    while (InBytes < CompSize) {
  115.       BitMap = (BYTE) fgetc(HelpFile);
  116.       NumToRead = BytesToRead(BitMap);
  117.  
  118.       if ((CompSize - InBytes) < NumToRead) 
  119.           NumToRead = CompSize - InBytes;   // only read what we have left
  120.       fread(Set, 1, (int) NumToRead, HelpFile);    
  121.       InBytes += NumToRead + 1;
  122.  
  123.       /* Go through and decode data */
  124.       for (counter = 0, Index = 0; counter < 8; counter++) {
  125.          /* It's a code, so decode it and copy the data */
  126.          if (BIT_SET(BitMap, counter)) {
  127.             Length = ((Set[Index+1] & 0xF0) >> 4) + 3;
  128.             Distance = (256 * (Set[Index+1] & 0x0F)) + Set[Index] + 1;
  129.             CodePtr = CurrPos - Distance;   // ptr into decompress window
  130.             while (Length)
  131.                { *CurrPos++ = *CodePtr++; OutBytes++; Length--; } 
  132.             Index += 2;  /* codes are 2 bytes */
  133.          }
  134.          else 
  135.             { *CurrPos++ = Set[Index++]; OutBytes++; }
  136.       }
  137.    }
  138.    return OutBytes;
  139. } 
  140.    
  141. // Prints a Phrase from the Phrase table
  142. void PrintPhrase(char *Phrases, int PhraseNum) {
  143.     int *Offsets = (int *)Phrases;
  144.     char *p = Phrases+Offsets[PhraseNum];
  145.     while (p < Phrases + Offsets[PhraseNum + 1])
  146.         { putchar(*p); p++; }
  147. }
  148.  
  149. // Build up a table of phrases
  150. void PhrasesLoad(FILE *HelpFile, long FileStart) {
  151.    FILEHEADER      FileHdr;
  152.    PHRASEHDR       PhraseHdr;
  153.    int             *Offsets;
  154.    char            *Phrases;
  155.    long            DeCompSize;
  156.  
  157.    /* Go to the phrases file and get the headers */
  158.    fseek(HelpFile, FileStart, SEEK_SET);
  159.    fread(&FileHdr, sizeof(FileHdr), 1, HelpFile);
  160.    fread(&PhraseHdr, sizeof(PhraseHdr), 1, HelpFile);
  161.  
  162.    /* Allocate space and decompress if it's compressed, else read in. */
  163.    if (Compressed) {
  164.       if ((Offsets = malloc((unsigned) (PhraseHdr.PhrasesSize + 
  165.           (PhraseHdr.NumPhrases + 1) * 2))) == NULL)
  166.         MSG("No room to decompress |Phrases");
  167.       Phrases = Offsets + fread(Offsets,2,PhraseHdr.NumPhrases+1, HelpFile);
  168.       DeCompSize = Decompress(HelpFile, (long)FileHdr.FileSize - 
  169.           (sizeof(PhraseHdr) + 2 * (PhraseHdr.NumPhrases+1)), Phrases);
  170.       if (DeCompSize != PhraseHdr.PhrasesSize) {
  171.          printf("\n");
  172.       }
  173.    }
  174.    else {
  175.       if (!(Offsets=malloc((unsigned)(FileHdr.FileSize-sizeof(PhraseHdr)))))
  176.          MSG("No room to decompress |Phrases");
  177.       /* Backup 4 bytes for uncompressed Phrases (no PhrasesSize) */
  178.       fseek(HelpFile, -4, SEEK_CUR);
  179.       fread(Offsets, (unsigned) (FileHdr.FileSize - 4), 1, HelpFile);
  180.    }
  181.    PhrasesPtr = Phrases = (char *) Offsets;
  182. }
  183.  
  184. /* Because the topic file is broken into 4k blocks, we'll have to handle
  185. all the reads.  The idea is to filter out the TOPICBLOCKHEADERs and
  186. do any decompression that needs doing. */
  187. long TopicRead(BYTE *Dest, long NumBytes, FILE *HelpFile) {
  188.    static long        CurrBlockLoc = 0;   /* Where we are in the block  */
  189.    static BYTE        *DCmpBlock = NULL;  /* Block of uncompressed data */
  190.    static long        DecompSize;         /* Size of block after decomp */
  191.    static long        TopicStart, BlkNum; /* Start of |TOPIC file       */
  192.    long               BytesLeft;          /* # Bytes left to return     */
  193.    TOPICBLOCKHEADER   BlockHeader;
  194.    TOPICLINK          *TempLink;
  195.    long               EndOffset;
  196.  
  197.    /* If NumBytes = 0, then we're done and need to free memory */
  198.    if (NumBytes == -1) { free(DCmpBlock); return 0; }
  199.  
  200.    if (!DCmpBlock) {
  201.       if (Compressed) {
  202.          if (! (DCmpBlock = malloc((unsigned) (4 * TopicBlockSize))))
  203.              FAIL("Not enough memory to decompress |TOPIC file");
  204.          TopicStart = ftell(HelpFile);
  205.          BlkNum = 0;
  206.       }
  207.       else if (! (DCmpBlock = malloc((unsigned) TopicBlockSize)))
  208.           FAIL("Not enough memory to handle |TOPIC file");
  209.       DecompSize = 0;   /* Set initial size to 0 */
  210.       /* Don't really need first block header, so get it out of the way */
  211.       fread(&BlockHeader, sizeof(BlockHeader), 1, HelpFile);
  212.    }
  213.  
  214.    BytesLeft = NumBytes;
  215.    while (BytesLeft) {
  216.       if (DecompSize == CurrBlockLoc) {
  217.          BlkNum++;
  218.  
  219.          if (Compressed) {
  220.             DecompSize = Decompress(HelpFile, (long)TopicBlockSize-1, 
  221.                 (char *)DCmpBlock);
  222.             /* Align ourselves at next 4k block */
  223.             fseek(HelpFile, TopicStart + (4096L * BlkNum), SEEK_SET);
  224.          }
  225.          else
  226.             DecompSize=fread(DCmpBlock,1,(unsigned) TopicBlockSize, HelpFile);
  227.  
  228.          CurrBlockLoc = 0;
  229.          fread(&BlockHeader, sizeof(BlockHeader), 1, HelpFile);
  230.  
  231.          // Get offset of last topic link. (Don't need block #, hence 3FFFh)
  232.          EndOffset = BlockHeader.LastTopicLink & 0x3FFF;
  233.          TempLink = (TOPICLINK*)(DCmpBlock + EndOffset-sizeof(BlockHeader));
  234.  
  235.          /* Actual end of the data (Don't include header) */
  236.          EndOffset += (TempLink->BlockSize - sizeof(BlockHeader));
  237.  
  238.          // If end shorter than topic block use it; else topic block full
  239.          if (EndOffset > DecompSize) {
  240.              /* Adjust DecompSize if crossing 4k boundary */
  241.              EndOffset = TempLink->BlockSize-((TempLink->NextBlock) & 0x3FFF);
  242.              DecompSize = (BlockHeader.LastTopicLink & 0x3FFF) + EndOffset;
  243.          }
  244.          else DecompSize = EndOffset;
  245.      } /* If */
  246.  
  247.      *(Dest++) = *(DCmpBlock + (CurrBlockLoc++) );
  248.       BytesLeft--;
  249.    } /* While (BytesLeft) */
  250.    return NumBytes;
  251. }
  252.  
  253. // Displays a string from a topic link record. Checks for Phrase
  254. // replacement and non-printable chars
  255. void TopicStringPrint(char *String, long Length) {
  256.    BYTE            Byte1, Byte2;
  257.    int             CurChar, PhraseNum;
  258.    long            counter;
  259.  
  260.    for (counter = 0; counter < Length; counter++) {
  261.       CurChar = * ((char *) (String + counter));
  262.  
  263.       /* Check for Phrase replacement! */
  264.       if ((CurChar > 0) && (CurChar < 10)) {
  265.          Byte1 = (BYTE) CurChar;
  266.          counter++;
  267.          CurChar = * ((char *) (String + counter));
  268.          Byte2 = (BYTE) CurChar;
  269.          PhraseNum = (256 * (Byte1 - 1) + Byte2);
  270.  
  271.          /* If there's a remainder, we have a space after the phrase */
  272.          PrintPhrase(PhrasesPtr, PhraseNum / 2);
  273.          if (PhraseNum % 2) putchar(' ');
  274.       }
  275.       else if (isprint(CurChar)) putchar(CurChar);
  276.       else putchar(' ');    // could do newline for 0x00 0x00
  277.    }
  278. }
  279.  
  280. // Dump |TOPIC file, doing decompression and phrase substitution
  281. void TopicDump(FILE *HelpFile, long FileStart) {
  282.    FILEHEADER      FileHdr;
  283.    TOPICHEADER     *TopicHdr;
  284.    TOPICLINK       TopicLink;
  285.  
  286.    /* Go to the TOPIC file and get the headers */
  287.    fseek(HelpFile, FileStart, SEEK_SET);
  288.    fread(&FileHdr, sizeof(FileHdr), 1, HelpFile);
  289.  
  290.    do {
  291.       TopicRead((BYTE *) &TopicLink, sizeof(TopicLink) - 4, HelpFile);
  292.  
  293.       if (Compressed)
  294.          TopicLink.DataLen2 = TopicLink.BlockSize - TopicLink.DataLen1;
  295.  
  296.       TopicLink.LinkData1=(BYTE *) malloc((unsigned)(TopicLink.DataLen1-21));
  297.       if(!TopicLink.LinkData1)
  298.           MSG("Error allocating TopicLink.LinkData1");
  299.       TopicRead(TopicLink.LinkData1, TopicLink.DataLen1 - 21, HelpFile);
  300.       if (TopicLink.DataLen2 > 0) {
  301.           TopicLink.LinkData2=(BYTE*)malloc((unsigned)(TopicLink.DataLen2+1));
  302.           if(!TopicLink.LinkData2)
  303.              MSG("Error allocating TopicLink.LinkData2");
  304.           TopicRead(TopicLink.LinkData2, TopicLink.DataLen2, HelpFile);
  305.       }
  306.  
  307.       /* Display a Topic Header record */
  308.       if (TopicLink.RecordType == TL_TOPICHDR) {
  309.          TopicHdr = (TOPICHEADER *)TopicLink.LinkData1;
  310.          printf("================ Topic Block Data ====================\n");
  311.          printf("Topic#: %ld - ", TopicHdr->TopicNum);
  312.  
  313.          if (TopicLink.DataLen2 > 0)
  314.             TopicStringPrint(TopicLink.LinkData2, (long) TopicLink.DataLen2);
  315.          else printf("\n");
  316.       }
  317.  
  318.       /* Show a 'text' type record. */
  319.       else if (TopicLink.RecordType == TL_DISPLAY) {
  320.          printf("-- Topic Link Data\n");
  321.          TopicStringPrint(TopicLink.LinkData2, (long) TopicLink.DataLen2);
  322.       }
  323.       printf("\n\n");
  324.       free(TopicLink.LinkData1);
  325.       if (TopicLink.DataLen2 > 0) free(TopicLink.LinkData2);
  326.    } while(TopicLink.NextBlock != -1);
  327. }
  328.  
  329. void DumpFile(FILE *HelpFile) {
  330.     long    FileOffset, PhraseOffset, TopicOffset;
  331.     DWORD   i;
  332.     char    FileName[32];
  333.  
  334.     fread(&HelpHeader, sizeof(HelpHeader), 1, HelpFile);
  335.     if (HelpHeader.MagicNumber != 0x35F3FL)
  336.         MSG("Fatal Error:  Not a valid WinHelp file");
  337.     WHIFSGetFirstLeaf(HelpFile);
  338.     TopicOffset = PhraseOffset = 0;
  339.  
  340.     for (i=0; i<WHIFSHeader.TotalWHIFSEntries; i++) {
  341.        GetFile(HelpFile, i, &FileOffset, FileName);
  342.        if (! strcmp(FileName, "|SYSTEM")) SysLoad(HelpFile, FileOffset);
  343.        else if (! strcmp(FileName, "|Phrases")) PhraseOffset = FileOffset;
  344.        else if (! strcmp(FileName, "|TOPIC")) TopicOffset = FileOffset;
  345.        }
  346.        if (PhraseOffset) PhrasesLoad(HelpFile, PhraseOffset);
  347.        if (TopicOffset) TopicDump(HelpFile, TopicOffset);
  348.        else MSG("No Topic file found!");
  349. }
  350.  
  351. int main(int argc, char *argv[]) {
  352.     char filename[40];
  353.     FILE *HelpFile;
  354.  
  355.     if (argc < 2) { 
  356.        printf("Usage: TOPICDMP helpfile[.hlp]\n\n");
  357.        printf("   helpfile      - Name of help file (.HLP or .MVB)\n\n");
  358.        return EXIT_FAILURE;
  359.        }
  360.  
  361.     if (! strchr(strcpy(filename, strupr(argv[1])), '.'))
  362.        strcat(filename, ".HLP");
  363.  
  364.     if ((HelpFile = fopen(filename, "rb")) == NULL) {
  365.        printf("Can't open %s!", filename);
  366.        return EXIT_FAILURE;
  367.     }
  368.  
  369.     DumpFile(HelpFile);
  370.     fclose(HelpFile);
  371.     return EXIT_SUCCESS;
  372. }
  373.  
  374.